# 导入工具包
import fasttext
from ft01_config import Config

conf = Config()

# 1、模型训练
model = fasttext.train_supervised(
    input='./final_data/train_fastText_char.txt',
)
#2、模型保存
path = conf.ft_model_save_path
model.save_model(path+"./fastText_char_default.bin")

#3、模型预测
print(model.predict("《 赤 壁 O L 》 攻 城 战 诸 侯 战 硝 烟 又 起"))

# 4、模型词表查看
print(f'查看模型词表[:10]：{model.words[:10]}')

# 5、查看模型子词，上述训练未开启子词，所以这里查到还是词本身
print(model.get_subwords('你好'))

# 6、模型测试评估
res = model.test('./final_data./test_fastText_char.txt')
#参考官方说明：https://fasttext.cc/docs/en/supervised-tutorial.html
print(res)  # (10000, 0.8761, 0.8761) 样本量 精确率 召回率

